import os

# 自定义目录大小统计函数
def getdirsize(dir):
    dlist = os.listdir(dir)
    sum = 0  # 总大小
    for f in dlist:
        file = os.path.join(dir, f)
        if os.path.isfile(file):  # 检查是文件还是文件夹
            sum += os.path.getsize(file)
        elif os.path.isdir(file):
            sum += getdirsize(file)  # 递归调用统计函数
    return sum


# 测试
path = input("请输入文件夹的绝对路径：")
if os.path.isdir(path):
    print("文件夹大小为：{}字节".format(getdirsize(path)))
else:
    print("不是文件夹，请检查名称或路径！")